package lc20240529;

import lc20240526.ListNode;

/**
 * @author 不知名帅哥
 * @version 1.0
 * @description 给你一个链表的头节点 head 和一个特定值 x ，
 * 请你对链表进行分隔，使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
 * 你应当 保留 两个分区中每个节点的初始相对位置。
 * @date 2024/5/29 22:13
 */
public class PartitionList {
    public static void main(String[] args) {
        ListNode head = new ListNode(1, new ListNode(4, new ListNode(3, new ListNode(2, new ListNode(5, new ListNode(2))))));

        ListNode res = partition(head, 3);
        while (res != null) {
            System.out.print(res.val + " ");
            res = res.next;
        }
    }

    //分成两个部分。连接两个链表。
    public static ListNode partition(ListNode head, int x) {
        // if (head == null || head.next == null) {
        //     return head;
        // }
        ListNode smallDummy = new ListNode(0);
        ListNode largeDummy = new ListNode(0);

        ListNode small = smallDummy;
        ListNode large = largeDummy;

        ListNode cur = head;
        while (cur != null) {
            if (cur.val < x) {
                small.next = cur;
                small = small.next;
            } else {
                large.next = cur;
                large = large.next;
            }
            cur = cur.next;
        }
        //拼接
        small.next = largeDummy.next;
        //置空，防止出现环。
        large.next = null;
        return smallDummy.next;
    }
}
